package com.jiang.utils;

import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.stereotype.Component;

import java.time.LocalDateTime;
import java.time.ZoneOffset;
import java.time.format.DateTimeFormatter;

/**
 * @author Jiang
 * @version 1.0
 * @date 2024/05/12 11:06
 */
@Component
public class RedisIdWorker {
    /**
     * 起始时间
     */
    private static final Long BEGIN_TIMESTAMP = 1704067200L;
    /**
     * 序列号位数
     */
    private static final Long COUNT_BITS = 32L;

    private StringRedisTemplate stringRedisTemplate;

    public RedisIdWorker(StringRedisTemplate stringRedisTemplate) {
        this.stringRedisTemplate = stringRedisTemplate;
    }

    // keyPrefix: 区分不同业务，使用不同的自增长
    public long nextId(String keyPrefix) {
        // 1、获取当前时间戳
        LocalDateTime now = LocalDateTime.now();
        long nowSecond = now.toEpochSecond(ZoneOffset.UTC);
        long timestamp = nowSecond - BEGIN_TIMESTAMP;
        // 2、生成序列号
        // 2.1.获取当前日期，精确到天（redis也有上限 2^64 ， 这里只能接收2^32， 所以不能一直增，加个日期，每天的key都不一样， 还有统计一天销售量的效果）
        String date = now.format(DateTimeFormatter.ofPattern("yyyy:MM:dd")); // redis的可以就是用 : 分隔， 统计月就可以yyyy:MM
        // 2.2.自增长 （即使到新的一天，key 不存在，increment方法会自动创建key 值为1， 不会有空指针）
        long count = stringRedisTemplate.opsForValue().increment("icr:" + keyPrefix + ":" + date);
        // 3、拼接并返回

        return timestamp << COUNT_BITS | count;
    }

    public static void main(String[] args) {
        LocalDateTime localDateTime = LocalDateTime.of(2024, 1, 1, 0, 0,0);
        long second = localDateTime.toEpochSecond(ZoneOffset.UTC);
        System.out.println(second);
    }
}
